package listnode;

import javax.management.DynamicMBean;
import java.util.HashMap;
import java.util.Map;

public class _1171_从链表中删去总和值为零的连续节点 {


    /**
     * 给你一个链表的头节点 head，请你编写代码，反复删去链表中由 总和 值为 0 的连续节点组成的序列，直到不存在这样的序列为止。
     * <p>
     * 删除完毕后，请你返回最终结果链表的头节点。
     * <p>
     *  
     * <p>
     * 你可以返回任何满足题目要求的答案。
     * <p>
     * 来源：力扣（LeetCode）
     * 链接：https://leetcode-cn.com/problems/remove-zero-sum-consecutive-nodes-from-linked-list
     * 著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。
     *
     * @param head
     * @return
     */
    public static ListNode removeZeroSumSublists(ListNode head) {
        ListNode dummy = new ListNode(0);
        dummy.next = head;

        Map<Integer, ListNode> map = new HashMap<>();

        // 首次遍历建立 节点处链表和<->节点 哈希表
        // 若同一和出现多次会覆盖，即记录该sum出现的最后一次节点
        int sum = 0;
        for (ListNode d = dummy; d != null; d = d.next) {
            sum += d.val;
            map.put(sum, d);
        }

        // 第二遍遍历 若当前节点处sum在下一处出现了则表明两结点之间所有节点和为0 直接删除区间所有节点
        sum = 0;
        for (ListNode d = dummy; d != null; d = d.next) {
            sum += d.val;
            d.next = map.get(sum).next;
        }

        return dummy.next;

    }

    public static void main(String[] args) {
        ListNode n1 = new ListNode(1);
        ListNode n2 = new ListNode(-2);
        ListNode n3 = new ListNode(2);
        ListNode n4 = new ListNode(3);
        ListNode n5 = new ListNode(-4);
        ListNode n6 = new ListNode(4);
        n1.next = n2;
        n2.next = n3;
        n3.next = n4;
        n4.next = n5;
        n5.next = n6;
        ListNode node = removeZeroSumSublists(n1);
    }

    public static ListNode removeZeroSumSublists1(ListNode head) {
        HashMap<Integer, ListNode> map = new HashMap<>();
        int sum = 0;
        ListNode dummy = new ListNode();
        dummy.next = head;
        ListNode pre = dummy;
        while (pre!=null){
            sum+=pre.val;
            map.put(sum, pre);
            pre = pre.next;
        }

        sum = 0;
        for (ListNode node = dummy; node!=null; node = node.next){
            sum += node.val;
            node.next = map.get(sum).next;
        }
        return dummy.next;
    }

}
